#region Copyright
// 
// Copyright (C) 2010 VirtualStaticVoid <virtualstaticvoid@gmail.com>
// 
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
//
#endregion

namespace IoCWrap.AutofacImpl
{
  using System;
  using Autofac;
  using Autofac.Builder;

  public class AutofacContainerBuilder : IContainerBuilder
  {

    private readonly ContainerBuilder _containerBuilder;

    public AutofacContainerBuilder()
    {
      this._containerBuilder = new ContainerBuilder();
      this._containerBuilder.RegisterSource(new UnregisteredTypesRegistrationSource());
    }

    public IoCWrap.IContainer Build()
    {
      return new AutofacContextWrapper(this._containerBuilder.Build());
    }

    public IContainerRegistration Register(Type type)
    {
      return new AutofacComponentRegistration<object, ConcreteReflectionActivatorData, SingleRegistrationStyle>
        (
          this._containerBuilder.RegisterType(type)
        );
    }

    public IContainerRegistration Register<TType>()
    {
      return new AutofacComponentRegistration<TType, ConcreteReflectionActivatorData, SingleRegistrationStyle>
        (
          this._containerBuilder.RegisterType<TType>()
        );
    }

    public IContainerRegistration Register<TType>(TType instance)
      where TType : class
    {
      return new AutofacComponentRegistration<TType, SimpleActivatorData, SingleRegistrationStyle>
        (
          this._containerBuilder.RegisterInstance(instance)
        );
    }

    public IContainerRegistration Register<TType>(ContainerComponentActivator<TType> activator)
    {
      return new AutofacComponentRegistration<TType, SimpleActivatorData, SingleRegistrationStyle>
        (
          this._containerBuilder.Register(container => activator(new AutofacContextWrapper(container)))
        );
    }

  }
}
